package com.birdboot.http;

import javax.activation.MimetypesFileTypeMap;
import java.io.*;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

/*
* V7新增的内容：
* 响应对象
* 该类的每一个实例用于表示服务器给客户端发送的一个HTTP的响应
* HTTP协议要求响应应由三部分构成：状态行，响应头，响应正文
* */
public class HttpServletResponse {
    private static MimetypesFileTypeMap mtfm = new MimetypesFileTypeMap();

    private Socket socket;
    //状态行相关信息
    private int statusCode = 200; //状态代码
    private String statusReason = "OK"; //状态描述
    //响应头相关信息
    private Map<String,String> headers = new HashMap<>();


    //响应正文相关信息
    private File contentFile; //相应正文对应的实体文件

    public HttpServletResponse(Socket socket){
        this.socket = socket;
    }
    public void response() throws IOException {
        //3.1 发送状态行
        sendStatusLine();
        //3.2 发送响应头
        sendHeaders();
        //3.3 发送响应正文
        sendContent();
    }
    //发送状态行
    private void sendStatusLine() throws IOException {
        //3.1 发送状态行
        println("HTTP/1.1" + " " + statusCode + " " + statusReason);
    }
    //发送响应头
    private void sendHeaders() throws IOException {
        //3.2 发送响应头
        /*
        * 遍历headers的所有待发送的响应头发送给浏览器
        * */
        Set<Map.Entry<String,String>> entrySet = headers.entrySet();
        for (Map.Entry<String,String> e :entrySet){
            String name = e.getKey();
            String value = e.getValue();
            println(name+":"+value);
        }
        //单独发送回车换行 表示响应头发送完毕
        println("");
    }
    //发送响应正文
    private void sendContent() throws IOException {
        //3.3 发送响应正文
        FileInputStream fis = new FileInputStream(contentFile);
        OutputStream out = socket.getOutputStream();
        byte[] buf = new byte[1024*10];
        int d;
        while ((d=fis.read(buf))!=-1){
            out.write(buf,0,d);
        }
    }
    /*
    V7:将ClientHandler中发送响应的工作全部移动到这里，println方法也是
    向客户发送一行字符串*/
    private void println(String line) throws IOException {
        OutputStream out = socket.getOutputStream();
        byte data[] = line.getBytes(StandardCharsets.ISO_8859_1);
        out.write(data);
        out.write(13);
        out.write(10);
    }

    public int getStatusCode() {
        return statusCode;
    }

    public void setStatusCode(int statusCode) {
        this.statusCode = statusCode;
    }

    public String getStatusReason() {
        return statusReason;
    }

    public void setStatusReason(String statusReason) {
        this.statusReason = statusReason;
    }

    public File getContentFile() {
        return contentFile;
    }
    /*
    * 设置响应正文对应的实体文件，该方法中会自动根据该文件添加对应的两个响应头
    * Content-Type和Content-Length
    * */
    public void setContentFile(File contentFile) {
        this.contentFile = contentFile;
        addHeader("Content-Type",mtfm.getContentType(contentFile));
        addHeader("Content-Length",contentFile.length()+"");
    }
    /*
    * 添加一个响应头
    * */
    public void addHeader(String name,String value){
        headers.put(name,value);
    }
}
